Micron Document
Fox's Git Mirrors

Node / RFnexus / code.git / commits / 41b8aba

Commit 41b8abaf83582bebefad298ca2505018ffb09ce6

🗀 Browse tree at this commit

Parents : f1aae95
Author : Ahmet Inan <inan@aicodix.de>
Date : 2024-02-14T10:48:45+01:00

added more shuffle operations for AVX2

Changes

2 files changed, 30 insertions(+), 0 deletions(-)

M avx2.hh +8
M avx2_double.hh +22

Diff

diff --git a/avx2.hh b/avx2.hh
index 4e49303..b275cee 100644
--- a/avx2.hh
+++ b/avx2.hh
@@ -1178,6 +1178,14 @@ inline SIMD<int8_t, 32> vshuf(SIMD<int8_t, 32> a, SIMD<uint8_t, 32> b)
return tmp;
}
+template <>
+inline SIMD<uint32_t, 8> vshuf(SIMD<uint32_t, 8> a, SIMD<uint32_t, 8> b)
+{
+ SIMD<uint32_t, 8> tmp;
+ tmp.m = _mm256_permutevar8x32_epi32(a.m, b.m);
+ return tmp;
+}
+
template <>
inline SIMD<float, 8> vshuf(SIMD<float, 8> a, SIMD<uint32_t, 8> b)
{

diff --git a/avx2_double.hh b/avx2_double.hh
index cb96cf7..cd01530 100644
--- a/avx2_double.hh
+++ b/avx2_double.hh
@@ -1321,3 +1321,25 @@ inline SIMD<int8_t, 64> vshuf(SIMD<int8_t, 64> a, SIMD<uint8_t, 64> b)
return tmp;
}
+template <>
+inline SIMD<uint32_t, 16> vshuf(SIMD<uint32_t, 16> a, SIMD<uint32_t, 16> b)
+{
+ SIMD<uint32_t, 16> tmp;
+ for (int i = 0; i < 2; ++i)
+ tmp.m[i] = _mm256_or_si256(
+ _mm256_and_si256(_mm256_permutevar8x32_epi32(a.m[0], b.m[i]), _mm256_cmpgt_epi32(_mm256_set1_epi32(8), b.m[i])),
+ _mm256_and_si256(_mm256_permutevar8x32_epi32(a.m[1], _mm256_sub_epi32(b.m[i], _mm256_set1_epi32(8))), _mm256_cmpgt_epi32(b.m[i], _mm256_set1_epi32(7))));
+ return tmp;
+}
+
+template <>
+inline SIMD<float, 16> vshuf(SIMD<float, 16> a, SIMD<uint32_t, 16> b)
+{
+ SIMD<float, 16> tmp;
+ for (int i = 0; i < 2; ++i)
+ tmp.m[i] = _mm256_or_ps(
+ _mm256_and_ps(_mm256_permutevar8x32_ps(a.m[0], b.m[i]), _mm256_cmpgt_epi32(_mm256_set1_epi32(8), b.m[i])),
+ _mm256_and_ps(_mm256_permutevar8x32_ps(a.m[1], _mm256_sub_epi32(b.m[i], _mm256_set1_epi32(8))), _mm256_cmpgt_epi32(b.m[i], _mm256_set1_epi32(7))));
+ return tmp;
+}
+

Served by rngit 1.4.1 - Generated in 0.04s